# 用户注册和登录
def loginAndAdd(i):
    if i == True:
        isLogin = 2
    else:
        isLogin = int(input("请选择你的选项：1.用户注册；2.用户登录；3退出\n"))
    if isLogin == 3:
        print("程序已退出=============")
        return False
    if isLogin == 1:
        alertStr = "请输入用户名：\n"
        while (True):
            userName = input(alertStr)
            if users.get(userName) == None:
                userPwd = input("请输入密码：\n")
                # global users
                users[userName] = userPwd
                print("注册成功，现在进行登录===")
                loginAndAdd(True)
                break
            else:
                alertStr = "用户名重复，请重新输入：\n"
    if isLogin == 2:
        userName = input("请输入用户名：\n")
        userPwd = input("请输入密码：\n")
        if users.get(userName):
            if userPwd == users.get(userName):
                print("欢迎" + userName + "用户，登录成功！")


users = {}
loginAndAdd(False)
